# Please install OpenAI SDK first: `pip3 install openai`
from openai import OpenAI
# 填写你的 API Key
API_KEY = "sk-12c657aa3a4f4f2daf43ad4ee0b8b1b8"  

def get_deepseek_response(question, api_key=API_KEY, 
                         system_prompt=None, model="deepseek-chat"):
    """
    调用DeepSeek API获取回答
    
    参数:
        question (str): 用户问题
        api_key (str): DeepSeek API密钥，默认使用预设值
        system_prompt (str): 系统提示，若为None则使用默认提示
        model (str): 使用的模型，默认为"deepseek-chat"
        
    返回:
        str: 模型回答内容
    """
    # 默认系统提示
    if system_prompt is None:
        system_prompt = """
        你是一个严谨的AI助手，必须用中文回答问题。
        你的回答必须详细，准确，并且遵循以下格式：
        如果问题需要实时信息，请使用联网搜索功能获取最新数据。
        """

    client = OpenAI(api_key=api_key, base_url="https://api.deepseek.com/v1")

    try:
        response = client.chat.completions.create(
            model=model,
            messages=[
                {"role": "system", "content": system_prompt},
                {"role": "user", "content": question},
            ],
            stream=False
        )
        return response.choices[0].message.content
    except Exception as e:
        return f"调用API出错: {str(e)}"


if __name__ == "__main__":

    # 简单调用
    response = get_deepseek_response("成都市青羊区今天的天气如何？")
    print(response)
